import { ImageData } from './image'
import { Task } from './task'

// 数据集接口
export interface Dataset {
    id: string
    name: string
    description: string
    type: '2D' | '3D'
    tasks: Task[]
    images: ImageData[]
    imageCount: number
    size: number // 数据集大小（字节）
    tags?: string[]
    isPublic: boolean
    owner?: string
    createdAt: Date
    updatedAt: Date
}

// 数据集统计信息
export interface DatasetStats {
    totalImages: number
    totalSize: number
    imageFormats: Record<string, number>
    modalities: Record<string, number>
    lastUpdated: Date
}

// 数据集创建请求
export interface CreateDatasetRequest {
    name: string
    description: string
    type: '2D' | '3D'
    tags?: string[]
    isPublic?: boolean
}

// 数据集更新请求
export interface UpdateDatasetRequest {
    name?: string
    description?: string
    tags?: string[]
    isPublic?: boolean
}